import torchvision

trans_totensor = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])
train_dataset = torchvision.datasets.CIFAR10(root='../data/CIFAR10', transform=trans_totensor, train=True, download=True)
test_dataset = torchvision.datasets.CIFAR10(root='../data/CIFAR10', transform=trans_totensor, train=False, download=True)

print(type(train_dataset))
print(train_dataset.classes)
print(train_dataset[0])
img, class1 = train_dataset[0]
print(type(img))
print(img.shape)
print(train_dataset.classes[class1])
